import os  # os操作系统
import math

size = 0
path = input("请输入您要统计的文件夹的绝对路径(当前目录下情输入#): ")
if path == '#':
    path = '.'


# 自定义目录的复制
def attributes(mydir):
    global size
    files = os.listdir(mydir)  # 获得当前 硬盘目录中的所有文件
    # 获取目录中的所有的文件信息
    for f in files:
        file1 = os.path.join(mydir, f)
        if os.path.isfile(file1):
            size = size + os.path.getsize(file1)
        if os.path.isdir(file1):
            attributes(file1)


def count(mydir):
    attributes(mydir)
    if math.pow(1024, 2) > size >= 1024:
        print("文件大小：{:.2f}KB".format(size / 1024))
    elif math.pow(1024, 3) > size >= math.pow(1024, 2):
        print("文件大小：{:.2f}MB".format(size / math.pow(1024, 2)))
    elif size >= math.pow(1024, 3):
        print("文件大小：{:.2f}GB".format(size / math.pow(1024, 3)))


count(path)
